package com.gonsin.gview.logic.services.download;

import org.apache.commons.lang3.RandomStringUtils;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Component;

import java.io.File;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

/**
 * 专门管理下载单例
 *
 * 存放 一次性下载的内容，下载完 即 删除
 */
@Component
public class DownloadManager {

    /**
     * 这里存放的都是一次性的下载内容
     */
    private Map<String, Object> downloadInstance = new ConcurrentHashMap<>();

    public Object getDownloadItem(String downloadId) {
        Object item = downloadInstance.get(downloadId);

        // 每次获取之后就删除，保证是一次性的，，保证不会产生无用的内存
        if(item != null){
            downloadInstance.remove(downloadId);
        }
        return item;
    }

    /**
     *
     * @param downloadId
     */
    public void removeDownloadItem(String downloadId) {
        downloadInstance.remove(downloadId);
    }


    /**
     * 每小时情一次
     */
    @Scheduled(cron = "0 0 0/1 * * ?")
    public void cleanCache(){
        downloadInstance.clear();
    }


    /**
     * 将要下载的内容放到map中，等待前端调用下载。
     */
    public String putDownloadItem(Object purity) {
        String downloadId = RandomStringUtils.random(32, true, true);
        downloadInstance.put(downloadId, purity);
        return downloadId;
    }
}
